package com.atguigu.gmall.item.comp;

import lombok.extern.slf4j.Slf4j;
import org.apache.http.impl.client.DefaultRedirectStrategy;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultReactiveScriptExecutor;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;

import javax.swing.*;
import java.time.Duration;
import java.util.Arrays;
import java.util.UUID;

/**
 * @ClassName RedisDistLock
 * @Description TODO
 * @Author king
 * @Date 2022/11/10 10:43
 * @Version 1.0
 */
@Slf4j
@Component
public class RedisDistLock { //业务超长: 延长锁时间或续期

    @Autowired
    StringRedisTemplate redisTemplate;

    //占坑 + 过期时间 = 原子
    public String lock(){
        String token = UUID.randomUUID().toString();
        //提前设置过期时间
        boolean absent = redisTemplate.opsForValue().setIfAbsent("lock",token,Duration.ofSeconds(10));

        //加锁失败继续加,一直加到成功位置,阻塞式等锁
        while(!absent){
            //睡一会//免得cpu受不了
            try {
                Thread.sleep(20);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            //没加到锁,一直加
            absent = redisTemplate.opsForValue().setIfAbsent("lock", token, Duration.ofSeconds(10));
        }

        //加到锁,设置过期时间---避免停电
        //redisTemplate.expire("lock", Duration.ofSeconds(10));
        //业务时间长,可把锁时间设置长一点,也可自动续期
        return token;
    }


    //得值判断 + 删除 = 原子
    public void unlock(String token){
        //有可能删除别人的锁; 得锁判断 + 删除锁

        //问题
        //1,删锁 = 得锁的值进程判断 + 判断成功就删锁
        //导致: 判断是对的,删除命令很久以后才到达,redis把之前的key早删除换成另一个,命令到达就可以直接把刚换的删除

        //不删别人的锁
        //if (token.equals(redisTemplate.opsForValue().get("lock"))){
            //redisTemplate.delete("lock");
        //}else{
            //log.info("别人的锁,不删");
        //}

        //原子脚本,lua
        String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1]\n" +
                "then\n" +
                "    return redis.call(\"del\",KEYS[1])\n" +
                "else\n" +
                "    return 0\n" +
                "end\n";

        //返回1 删除成功
        //返回0 代表失败: 1), 别人的锁 2), 已经删除,没有这个key
        Long lock = redisTemplate.execute(new DefaultRedisScript<>(script,Long.class),
                Arrays.asList("lock"),token);

        if (lock>0){
            log.info("解锁" + token);
        }
        log.info("删锁成功");


    }

}
